package trastienda.dao;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;


import trastienda.excepcion.DAOExcepcion;
import trastienda.modelo.Servicio;
import trastienda.util.ConexionBD;


public class ServicioDAO extends BaseDAO {

	public Servicio obtener(int idServicios) throws DAOExcepcion {
		Servicio vo = new Servicio();
		Connection con = null;
		PreparedStatement stmt = null;
		ResultSet rs = null;
		try {
			String query = "select idServicios, nomServicio, precioServicio from Servicios where idServicios=?";
			con = ConexionBD.obtenerConexion();
			stmt = con.prepareStatement(query);
			stmt.setInt(1, idServicios);
			rs = stmt.executeQuery();
			if (rs.next()) {
				vo.setIdServicios(rs.getInt(1));
				vo.setNomServicio(rs.getString(2));
				vo.setPrecioServicio(rs.getInt(3));
			}
		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarResultSet(rs);
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
		return vo;
	}
	
	public Servicio insertar(Servicio vo) throws DAOExcepcion {
		String query = "insert into Servicios(nomServicio,precioServicio) values (?,?)";
		Connection con = null;
		PreparedStatement stmt = null;
		ResultSet rs = null;
		try {
			con = ConexionBD.obtenerConexion();
			stmt = con.prepareStatement(query);
			stmt.setString(1, vo.getNomServicio());
			stmt.setInt(2, vo.getPrecioServicio());
			int i = stmt.executeUpdate();
			if (i != 1) {
				throw new SQLException("No se pudo insertar");
			}
			// Obtener el ultimo id
			int id = 0;
			query = "select last_insert_id()";
			stmt = con.prepareStatement(query);
			rs = stmt.executeQuery();
			if (rs.next()) {
				id = rs.getInt(1);
			}
			vo.setIdServicios(id);

		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarResultSet(rs);
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
		return vo;
	}


}
